/*
 * @Author: your name
 * @Date: 2021-09-07 17:34:17
 * @LastEditTime: 2021-09-07 17:54:47
 * @LastEditors: your name
 * @Description: In User Settings Edit
 * @FilePath: \bigdate\src\ntils\cipher.ts
 */
import { encrypt, decrypt } from 'crypto-js/aes'
import UTF8, { parse } from 'crypto-js/enc-utf8'
import pkcs7 from 'crypto-js/pad-pkcs7'
import ECB from 'crypto-js/mode-ecb'
import md5 from 'crypto-js/md5'

import Base64 from 'crypto-js/enc-base64'

export interface EncryptionParams {
  key: string;
  iv: string;
}

export class AesEncryption {
    private key

    private iv

    constructor(opt: Partial<EncryptionParams> = {}) {
        const { key, iv } = opt
        if (key) {
            this.key = parse(key)
        }
        if (iv) {
            this.iv = parse(iv)
        }
    }

    get getOptions() {
        return {
            mode: ECB,
            padding: pkcs7,
            iv: this.iv,
        }
    }

    // 加密aes
    encryptByAES(cipherText: string) {
        return encrypt(cipherText, this.key, this.getOptions).toString()
    }

    // 解密
    decryptByAES(cipherText: string) {
        return decrypt(cipherText, this.key, this.getOptions).toString(UTF8)
    }
}
// 解码 base64
export function encryptByBase64(cipherText: string) {
    return UTF8.parse(cipherText).toString(Base64)
}
// 加密 base64
export function decodeByBase64(cipherText: string) {
    return Base64.parse(cipherText).toString(UTF8)
}

export function encryptByMd5(password: string) {
    return md5(password).toString()
}
